package labox.innovation.gameserver.pathfinding.utils;

import java.util.ArrayList;
import java.util.Map;

import javolution.util.FastMap;
import labox.innovation.gameserver.pathfinding.Node;

/**
 * 
 * @author Sami
 */
public class CellNodeMap {

	private Map<Integer, ArrayList<Node>> _cellIndex;

	public CellNodeMap() {
		_cellIndex = new FastMap<Integer, ArrayList<Node>>();
	}

	public void add(Node n) {
		if (_cellIndex.containsKey(n.getLoc().getY())) {
			_cellIndex.get(n.getLoc().getY()).add(n);
		} else {
			ArrayList<Node> array = new ArrayList<Node>(5);
			array.add(n);
			_cellIndex.put(n.getLoc().getY(), array);
		}
	}

	public boolean contains(Node n) {
		ArrayList<Node> array = _cellIndex.get(n.getLoc().getY());
		if (array == null) {
			return false;
		}
		for (Node node : array)
			if (node.equals(n))
				return true;
		return false;
	}
}
